package com.yunhe.dao;


// 员工表
import com.yunhe.pojo.Department;
import com.yunhe.pojo.Employee;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Repository;

import java.util.Collection;
import java.util.HashMap;
import java.util.Map;

@Repository
public class EmployeeDao {

    // 模拟数据库表中的数据
    private static Map<Integer, Employee> employees;

    @Autowired
    private DepartmentDao departmentDao;


    static {
        employees = new HashMap<>();

        employees.put(1001,new Employee(1001,"张三","111111@qq.ocm",1,new Department(101,"市场部")));
        employees.put(1002,new Employee(1002,"李四","222222@qq.ocm",0,new Department(102,"科研部")));
        employees.put(1003,new Employee(1003,"王五","333333@qq.ocm",1,new Department(103,"后勤部")));
        employees.put(1004,new Employee(1004,"王二","444444@qq.ocm",0,new Department(104,"销售部")));
        employees.put(1005,new Employee(1005,"麻子","555555@qq.ocm",1,new Department(105,"小卖部")));
    }

    // 主键自增
    private static Integer initId = 1006;

    // 添加一个员工
    public void save(Employee employee){
        if(employee.getId() == null){
            employee.setId(initId++);
        }

        employee.setDepartment(departmentDao.getDepartmentById(employee.getDepartment().getId()));
        employees.put(employee.getId(),employee);

    }

    // 查询全部员工信息
    public Collection<Employee> getAll(){
        return employees.values();
    }

    // 通过id查询员工
    public Employee getEmployeeById(Integer id){
        return employees.get(id);
    }

    // 通过id删除员工
    public void delete(Integer id){
        employees.remove(id);
    }


}
